TableRowTransferHandler.java

package jp.crwdev.app.util;

import java.awt.Cursor;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.dnd.DragSource;

import javax.activation.ActivationDataFlavor;
import javax.activation.DataHandler;
import javax.swing.JComponent;
import javax.swing.JTable;
import javax.swing.TransferHandler;
import javax.swing.table.DefaultTableModel;

import jp.crwdev.app.constant.Constant;
import jp.crwdev.app.interfaces.IImageFileInfo;
import jp.crwdev.app.interfaces.IImageFileInfoList;
import jp.crwdev.app.interfaces.ISimpleCallback;

@SuppressWarnings("serial")
public class TableRowTransferHandler extends TransferHandler {
	private int[] rows    = null;
	private int addIndex  = -1; //Location where items were added
	private int addCount  = 0;  //Number of items added.
	private final DataFlavor localObjectFlavor;
	private JComponent source = null;
	private IImageFileInfoList mList = null;
	private ISimpleCallback mCallback = null;
	
	public TableRowTransferHandler(IImageFileInfoList list, ISimpleCallback callback) {
		mList = list;
		mCallback = callback;
		localObjectFlavor = new ActivationDataFlavor(
				int[].class, DataFlavor.javaJVMLocalObjectMimeType, "Array of items");
	}
	
	@Override
	protected Transferable createTransferable(JComponent c) {
		source = c;
		JTable table = (JTable) c;
		rows = table.getSelectedRows();
		return new DataHandler(rows,localObjectFlavor.getMimeType());
	}
	
	@Override
	public boolean canImport(TransferSupport info) {
		JTable t = (JTable)info.getComponent();
		boolean b = info.isDrop()&&info.isDataFlavorSupported(localObjectFlavor);
		if(!info.isDataFlavorSupported(localObjectFlavor)){
			t.setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
		}else{
			//XXX bug?
			t.setCursor(b?DragSource.DefaultMoveDrop:DragSource.DefaultMoveNoDrop);
		}
		return b;
	}
	
	@Override
	public int getSourceActions(JComponent c) {
		return TransferHandler.MOVE;//COPY_OR_MOVE;
	}
	
	@Override
	public boolean importData(TransferSupport info) {
		JTable target = (JTable)info.getComponent();
		JTable.DropLocation dl = (JTable.DropLocation)info.getDropLocation();
		DefaultTableModel model = (DefaultTableModel)target.getModel();
		int index = dl.getRow();
		int max = model.getRowCount();
		if(index<0 || index>max) index = max;
		addIndex = index;
		target.setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
		try {
			int[] indices =
					(int[])info.getTransferable().getTransferData(localObjectFlavor);
			if(source==target) addCount = indices.length;
			for(int i=0;i<indices.length;i++) {
				int idx = index++;
				IImageFileInfo fileInfo = mList.get(indices[i]);
				model.insertRow(idx, Constant.createRecord(fileInfo));
				mList.insert(idx, fileInfo);
				target.getSelectionModel().addSelectionInterval(idx, idx);
			}
			return true;
		}catch(Exception ufe) { ufe.printStackTrace(); }
		return false;
	}
	
	@Override
	protected void exportDone(JComponent c, Transferable t, int act) {
		cleanup(c, act == MOVE);
	}
	
	private void cleanup(JComponent src, boolean remove) {
		if(remove && rows != null) {
			JTable table = (JTable)src;
			src.setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
			DefaultTableModel model = (DefaultTableModel)table.getModel();
			if(addCount > 0) {
				for(int i=0;i<rows.length;i++) {
					if(rows[i]>=addIndex) { rows[i] += addCount; }
				}
			}
			for(int i=rows.length-1;i>=0;i--){
				model.removeRow(rows[i]);
				mList.remove(rows[i]);
			}
			mList.setEnableSort(false);
			if(mCallback != null){
				mCallback.onCallback();
			}
		}
		rows     = null;
		addCount = 0;
		addIndex = -1;
	}
}